# 添加 cmake 模块


# 添加 C++ 源代码
file(GLOB EXT_PY_CPP_SRCS *.cpp)
add_library(PY_NPU_OBJ OBJECT ${EXT_PY_CPP_SRCS})

# 添加 include 目录
target_include_directories(PY_NPU_OBJ PUBLIC ${TORCH_INCLUDE_DIRS})

target_include_directories(PY_NPU_OBJ PUBLIC ${PYTHON_INCLUDE_DIR})
target_include_directories(PY_NPU_OBJ PUBLIC ${pybind11_INCLUDE_DIRS})